package com.linzm.leetcode.primary.exercises3_20230122;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * @Author zimingl
 * @Date 2023/1/23 14:12
 * @Description: 错误的集合
 */
public class Demo3_645 {
    /**
     * 集合 s 包含从 1 到 n 的整数。不幸的是，因为数据错误，导致集合里面某一个数字复制了成了集合里面的另外一个数字的值，导致集合 丢失了一个数字 并且 有一个数字重复 。
     * 给定一个数组 nums 代表了集合 S 发生错误后的结果。
     * 请你找出重复出现的整数，再找到丢失的整数，将它们以数组的形式返回。
     * 输入：nums = [1,2,2,4]
     * 输出：[2,3]
     */
    public static void main(String[] args) {
        Demo3_645 demo3_645 = new Demo3_645();
//        int[] nums = {1, 2, 2, 4};
//        int[] nums = {2, 3, 3, 4, 5, 6, 7};
//        int[] nums = {2, 2};
//        int[] nums = {1, 1};
        int[] nums = {3,2,2};
//        int[] nums = {2,3,2};
        int[] errorNums = demo3_645.findErrorNums2(nums);
        String[] strings = Arrays.stream(errorNums).mapToObj(String::valueOf).toArray(String[]::new);
        System.out.println(String.join(" ", strings));
    }

    // 1 2 3 3 4 6 7
    // 2 3 3 4 5 6 7
    private int[] findErrorNums(int[] nums) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int num : nums) {
            map.put(num, map.getOrDefault(num, 0) + 1);
        }
        int num = 0, diff = 0;
        for (int i = 1; i <= nums.length; i++) {
            int count = map.getOrDefault(i, 0);
            if (count == 2) {
                num = i;
            }
            if (count == 0) {
                diff = i;
            }
        }
        return new int[]{num, diff};
    }

    private int[] findErrorNums2(int[] nums) {
        int[] errorNums = new int[2];
        int n = nums.length;
        Arrays.sort(nums);
        int prev = 0;
        for (int i = 0; i < n; i++) {
            int curr = nums[i];
            if (curr == prev) {
                errorNums[0] = prev;
            } else if (curr - prev > 1) {
                errorNums[1] = prev + 1;
            }
            prev = curr;
        }
        if (nums[n - 1] != n) {
            errorNums[1] = n;
        }
        return errorNums;
    }
}
